D:\a\tools.proto\tools.proto\compiler\src\gen\codec.rs
Line | Count | Source |
1 | | // Copyright (c) 2024, BlockProject 3D |
2 | | // |
3 | | // All rights reserved. |
4 | | // |
5 | | // Redistribution and use in source and binary forms, with or without modification, |
6 | | // are permitted provided that the following conditions are met: |
7 | | // |
8 | | // * Redistributions of source code must retain the above copyright notice, |
9 | | // this list of conditions and the following disclaimer. |
10 | | // * Redistributions in binary form must reproduce the above copyright notice, |
11 | | // this list of conditions and the following disclaimer in the documentation |
12 | | // and/or other materials provided with the distribution. |
13 | | // * Neither the name of BlockProject 3D nor the names of its contributors |
14 | | // may be used to endorse or promote products derived from this software |
15 | | // without specific prior written permission. |
16 | | // |
17 | | // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
18 | | // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
19 | | // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
20 | | // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR |
21 | | // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
22 | | // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
23 | | // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
24 | | // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
25 | | // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING |
26 | | // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
27 | | // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
28 | | |
29 | | use crate::gen::template::Template; |
30 | | use std::collections::HashMap; |
31 | | |
32 | | macro_rules! count { |
33 | | () => (0usize); |
34 | | ( $x:tt $($xs:tt)* ) => (1usize + count!($($xs)*)); |
35 | | } |
36 | | |
37 | | pub(crate) use count; |
38 | | |
39 | | #[macro_export] |
40 | | macro_rules! codec_map_initializer { |
41 | | ($($name: literal => $template: expr),*) => { |
42 | | { |
43 | | use $crate::gen::codec::count; |
44 | | let mut map = std::collections::HashMap::with_capacity($crate::gen::codec::count!($($name)*)); |
45 | | $(map.insert($name, $crate::gen::template::Template::compile($template).unwrap());)* |
46 | | map |
47 | | } |
48 | | }; |
49 | | } |
50 | | |
51 | | pub struct CodecMap<'fragment, 'variable> { |
52 | | map: HashMap<&'fragment str, Template<'fragment, 'variable>>, |
53 | | } |
54 | | |
55 | | impl Default for CodecMap<'_, '_> { |
56 | 396 | fn default() -> Self { |
57 | 396 | Self::new(HashMap::new()) |
58 | 396 | } |
59 | | } |
60 | | |
61 | | impl<'fragment, 'variable> CodecMap<'fragment, 'variable> { |
62 | 399 | pub fn new(map: HashMap<&'fragment str, Template<'fragment, 'variable>>) -> Self { |
63 | 399 | Self { map } |
64 | 399 | } |
65 | | |
66 | 78 | pub fn has(&self, name: &str) -> bool { |
67 | 78 | self.map.contains_key(name) |
68 | 78 | } |
69 | | |
70 | 10 | pub fn insert(&mut self, name: &'fragment str, codec: Template<'fragment, 'variable>) { |
71 | 10 | self.map.insert(name, codec); |
72 | 10 | } |
73 | | |
74 | 289 | pub fn get(&self, name: &str) -> Option<&Template<'fragment, 'variable>> { |
75 | 289 | self.map.get(name) |
76 | 289 | } |
77 | | } |